export TASK_QUEUE_ENABLE=2
export CPU_AFFINITY_CONF=1

cd PoweFlowNet
torchrun --standalone --nproc_per_node=8 multiprocess.py --cfg_json ./configs/standard.json --num-epochs 2000 --data-dir ./data --batch-size 128 --train_loss_fn mse_loss --lr 0.001 --case 118v2 --model MaskEmbdMultiMPN --save